package com.zjj.lbw.ai.ollama;

import dev.langchain4j.model.ollama.OllamaEmbeddingModel;
import dev.langchain4j.model.openai.OpenAiEmbeddingModel;

/**
 * 测试基于ollama部署的本地向量模型
 */
public class ollamaEmbeddingTest {

    public static void main(String[] args) {
        OllamaEmbeddingModel model = OllamaEmbeddingModel.builder()
                .baseUrl("http://localhost:11434/")
                .modelName("nomic-embed-text:v1.5")
                .build();

        // ollama对向量模型兼容open-ai支持还不够好
//        OpenAiEmbeddingModel model = OpenAiEmbeddingModel.builder()
//                .baseUrl("http://localhost:11434/")
//                .apiKey("123")
//                .modelName("nomic-embed-text:v1.5")
//                .build();

        System.out.println(model.embed("你是谁").content());

    }
}
